How to solve index out of range error in python